# 1.准备数据

```
mkdir data

cd data 


ln -s ../../../datasets/DroneRGBT DroneRGBT
ln -s ../../../datasets/RGBTCC RGBTCrowdCounting


cd ..


python preprocess_RGBTCC.py
python preprocess_DroneRGBT.py

```

# 2.训练

```
wget https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window12_384_22k.pth


python train.py \
    --data-dir "./data/processedRGBTCC" \
    --save-dir "./pth_RGBTCC" \
    --dataset 'RGBTCC' \
    --lr 1e-5 \
    --device 0


python train.py \
    --data-dir "./data/processedDroneRGBT" \
    --save-dir "./pth_DroneRGBT" \
    --dataset 'DroneRGBT' \
    --lr 1e-5 \
    --device 0
```

# 3.测试

```
python test.py \
    --data-dir "./data/processedRGBTCC" \
    --save-dir "./pth_RGBTCC/0912-051539/best_model_0.pth" \
    --dataset 'RGBTCC' \
    --device 0


python test.py \
    --data-dir "./data/processedDroneRGBT" \
    --save-dir "./pth_DroneRGBT/0912-051701/best_model_0.pth" \
    --dataset 'DroneRGBT' \
    --device 0
```